package com.lhkj.ct.base.utils.sql;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.exceptions.UtilException;
import cn.hutool.core.util.ArrayUtil;
import com.lhkj.ct.base.mybatis.IBaseEntity;
import org.apache.commons.lang3.StringUtils;

import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * sql操作工具类
 */
@Deprecated
public class SqlUtil {
    /**
     * 定义常用的 sql关键字
     */
    public static String SQL_REGEX = "and |extractvalue|updatexml|exec |insert |select |delete |update |drop |count |chr |mid |master |truncate |char |declare |or |+|user()";

    /**
     * 仅支持字母、数字、下划线、空格、逗号、小数点（支持多个字段排序）
     */
    public static String SQL_PATTERN = "[a-zA-Z0-9_\\ \\,\\.]+";

    /**
     * 限制orderBy最大长度
     */
    private static final int ORDER_BY_MAX_LENGTH = 500;

    /**
     * 检查字符，防止注入绕过
     */
    public static String escapeOrderBySql(String value) {
        if (StringUtils.isNotEmpty(value) && !isValidOrderBySql(value)) {
            throw new UtilException("参数不符合规范，不能进行查询");
        }
        if (StringUtils.length(value) > ORDER_BY_MAX_LENGTH) {
            throw new UtilException("参数已超过最大限制，不能进行查询");
        }
        return value;
    }

    /**
     * 验证 order by 语法是否符合规范
     */
    public static boolean isValidOrderBySql(String value) {
        return value.matches(SQL_PATTERN);
    }

    /**
     * SQL关键字检查
     */
    public static void filterKeyword(String value) {
        if (StringUtils.isEmpty(value)) {
            return;
        }
        String[] sqlKeywords = StringUtils.split(SQL_REGEX, "\\|");
        for (String sqlKeyword : sqlKeywords) {
            if (StringUtils.indexOfIgnoreCase(value, sqlKeyword) > -1) {
                throw new UtilException("参数存在SQL注入风险");
            }
        }
    }

    /**
     * TODO 一对多合并数据
     *
     * @param newList       新数据
     * @param oldList       旧数据
     * @param insertOption  新增操作
     * @param updateOption  编辑操作
     * @param deleteOption  删除操作
     */
    public static <T extends IBaseEntity<String>> void merge(List<T> newList,
                                               List<T> oldList,
                                               Consumer<T> insertOption,
                                               Function<T, Integer> updateOption,
                                               Function<String[], Integer> deleteOption) {

        List<String> deleteIds = oldList.stream().map(T::getId).collect(Collectors.toList());
        Set<T> collect = new HashSet<>(oldList);
        for (T t : newList) {
            if (StringUtils.isBlank(t.getId())) {
                insertOption.accept(t);
                continue;
            }

            deleteIds.remove(t.getId());

            if (collect.contains(t)) {
                continue;
            }

            if (StringUtils.isNotBlank(t.getId())) {
                updateOption.apply(t);
            }
        }

        if (CollectionUtil.isNotEmpty(deleteIds)) {
            deleteOption.apply(ArrayUtil.toArray(deleteIds, String.class));
        }
    }
}
